package com.itcaius.weblog.admin.utils;

import com.itcaius.weblog.admin.config.MinioProperties;
import io.minio.MinioClient;
import io.minio.PutObjectArgs;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import java.util.UUID;

/**
 * @author Caius
 * @description
 * @since Created in 2025-04-26
 */
@Component
@Slf4j
public class MinioUtil {

    @Autowired
    private MinioProperties minioProperties;

    @Autowired
    private MinioClient minioClient;

    /**
     * 上传文件
     * @param file
     * @return
     * @throws Exception
     */
    public String uploadFile(MultipartFile file) throws Exception {
        // 文件判空
        if (file == null || file.getSize() == 0) {
            log.error("==> 上传文件异常：文件大小不能为空 ...");
            throw new RuntimeException("文件大小不能为空");
        }

        // 文件原始名称
        String originalFilename = file.getOriginalFilename();
        // 文件的 Context-Type
        String contentType = file.getContentType();

        // 生成存储对象的名称(将 uuid 字符串中的 - 替换为空字符串)
        String key = UUID.randomUUID().toString().replace("-", "");
        // 获取文件后缀
        String suffix = originalFilename.substring(originalFilename.lastIndexOf("."));

        // 拼接上文件后缀，为存储文件的名称
        String objectName = String.format("%s%s", key, suffix);

        log.info("==> 文件开始上传至 Minio, ObjetName: {}", objectName);

        minioClient.putObject(PutObjectArgs.builder()
                        .bucket(minioProperties.getBucketName())
                        .object(objectName)
                        .stream(file.getInputStream(), file.getSize(), -1)
                        .contentType(contentType)
                .build());

        // 返回文件访问路径
        String url = String.format("%s/%s/%s", minioProperties.getEndpoint(), minioProperties.getBucketName(), objectName);

        log.info("==> 文件上传至 Minio 成功, 访问路径： {}", url);

        return url;
    }
}
